import os
import time

import allure


"""
1.调用该函数，首页会截取移动端app当前页面，存放到手机指定路径下；
2.把手机端该路径下的图片通过adb拉取到项目指定的目录；
3.把该目录下的图片作为附件存放到allure报告中；
"""
def adb_shotScreen(funcName):
    currTime = time.strftime("%m%d%H%M%S")   # 获取当前时间的时间戳，例如：0219161200
    mobile_path = r'/sdcard/Pictures/'+ currTime + '.png'
    print(mobile_path)
    adb_shot = f"adb shell screencap {mobile_path}"
    file_dir = os.path.dirname(__file__)  # 获取当前__file__的文件目录(zip_file.py的上级目录）
    pc_path = os.path.join(file_dir,"..","screen_shot") # 通过join进行路径拼接
    print(f"pc_path的路径是：{pc_path}")
    adb_pull_shot = f"adb pull {mobile_path} {pc_path} "
    filename = pc_path +'\\'+ currTime + '.png'    # 截图在项目中的完整路径
    os.system(adb_shot)
    os.system(adb_pull_shot)
    allure.attach.file(source=filename, name=f'{funcName}+{currTime}', attachment_type=allure.attachment_type.PNG)


if __name__ == '__main__':
    currTime = time.strftime("%m%d%H%M%S")
    print(currTime)
    file_dir = os.path.dirname(__file__)   #C:/Users/37210/Desktop/测开班_16/app_ui_frame16/common
    print(file_dir)
    pc_path = os.path.join(file_dir, "..", "screen_shot")  # 通过join进行路径拼接
    print(pc_path)